package com.lianyu507.util;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

/**
 * Created by Peter on 2017/3/29.
 */
public class FileUtil {

    /**
     * 创建临时文件
     *
     * @param inputStream
     * @param name        文件名
     * @param ext         扩展名
     * @param tmpDirFile  临时文件夹目录
     */
    public static File createTmpFile(InputStream inputStream, String name, String ext, File tmpDirFile) throws IOException {
        File tmpFile;
        if (tmpDirFile == null) {
            tmpFile = File.createTempFile(name, '.' + ext);
        } else {
            tmpFile = File.createTempFile(name, '.' + ext, tmpDirFile);
        }

        tmpFile.deleteOnExit();

        try (FileOutputStream fos = new FileOutputStream(tmpFile)) {
            int read = 0;
            byte[] bytes = new byte[1024 * 100];
            while ((read = inputStream.read(bytes)) != -1) {
                fos.write(bytes, 0, read);
            }

            fos.flush();
            return tmpFile;
        }
    }

    /**
     * 创建临时文件
     *
     * @param inputStream
     * @param name        文件名
     * @param ext         扩展名
     */
    public static File createTmpFile(InputStream inputStream, String name, String ext) throws IOException {
        return createTmpFile(inputStream, name, ext, null);
    }
}
